package com.cg.leetcode;

import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * 54.螺旋矩阵
 *
 * @author cg
 * @program LeetCode->LeetCode_54
 * @create 2022-07-08 15:12
 **/
public class LeetCode_54 {

    @Test
    public void test54() {
        System.out.println(spiralOrder(new int[][]{{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}));
        System.out.println(spiralOrder(new int[][]{{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}}));
    }

    /**
     * 给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。
     * <p>
     * 示例 1：
     * 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
     * 输出：[1,2,3,6,9,8,7,4,5]
     * <p>
     * 示例 2：
     * 输入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
     * 输出：[1,2,3,4,8,12,11,10,9,5,6,7]
     * <p>
     * 提示：
     * m == matrix.length
     * n == matrix[i].length
     * 1 <= m, n <= 10
     * -100 <= matrix[i][j] <= 100
     *
     * @param matrix
     * @return
     */
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> res = new ArrayList<>();
        int m = matrix.length;
        int n = matrix[0].length;
        int mid = Math.min(m, n) / 2;
        int loop = 0;
        int start = 0;
        int i, j;
        while (loop++ < mid) {
            // 上侧从左至右
            for (j = start; j < n - loop; j++) {
                res.add(matrix[start][j]);
            }
            // 右侧从上至下
            for (i = start; i < m - loop; i++) {
                res.add(matrix[i][j]);
            }
            // 下侧从右至左
            for (; j > start; j--) {
                res.add(matrix[i][j]);
            }
            // 左侧从下至上
            for (; i > start; i--) {
                res.add(matrix[i][j]);
            }
            start++;
        }
        if (Math.min(m, n) % 2 == 1) {
            if (m > n) {
                for (int k = mid; k < mid + m - n + 1; k++) {
                    res.add(matrix[k][mid]);
                }
            } else {
                for (int k = mid; k < mid + n - m + 1; k++) {
                    res.add(matrix[mid][k]);
                }
            }
        }
        return res;
    }

}
